package com.zcx.excel.controller;

import com.alibaba.excel.EasyExcel;
import com.zcx.excel.common.vo.ApiResult;
import com.zcx.excel.listen.PriceListener;
import com.zcx.excel.pojo.Price;
import com.zcx.excel.pojo.PriceExample;
import com.zcx.excel.service.PriceService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;

/**
 * @author zhaochangxin
 * @Title: EasyExcelWebController
 * @Package com.zcx.excel.controller
 * @Description: 测试EasyExcel的web上传与下载
 * @date 2022/3/29 11:43
 */
@RestController
public class EasyExcelWebController {
    @Autowired
    PriceService priceService;
    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     * <p>1. 创建excel对应的实体对象
     * <p>2. 设置返回的 参数
     * <p>3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
     */
    @GetMapping("/download")
    public ApiResult download(PriceExample priceExample, HttpServletResponse response) throws IOException {
        PriceExample.Criteria criteria=priceExample.createCriteria();
        criteria.andCreatedByIsNotNull();
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("测试", "UTF-8");
        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), Price.class).sheet("模板").doWrite(priceService.queryAllPriceList(priceExample));
        return ApiResult.success();
    }

    /**
     * 文件上传
     * <p>1. 创建excel对应的实体对象
     * <p>2. 由于默认一行行的读取excel，所以需要创建excel一行一行的回调监听器
     * <p>3. 直接读即可
     */
    @PostMapping("upload")
    /*
    MultipartFile file
    传入参数
     */
    public String upload() throws IOException {
        String fileName ="D:\\Tesla\\test\\test1648452751022"+ ".xlsx";
        EasyExcel.read(fileName, Price.class, new PriceListener(priceService)).sheet().doRead();
        return "success";
    }


}
